python program for average calculation